package com.bdqn;

import org.w3c.dom.Document;
import org.w3c.dom.Element;
import org.w3c.dom.Text;
import org.xml.sax.SAXException;

import javax.xml.parsers.DocumentBuilder;
import javax.xml.parsers.DocumentBuilderFactory;
import javax.xml.parsers.ParserConfigurationException;
import javax.xml.transform.*;
import javax.xml.transform.dom.DOMSource;
import javax.xml.transform.stream.StreamResult;
import java.io.FileOutputStream;
import java.io.IOException;

/**
 * 向DOM中添加新节点
 */
public class SL03 {

    public static void main(String[] args) {
        // 得到DOM解析器的工厂实例
        DocumentBuilderFactory dbf = DocumentBuilderFactory.newInstance();
        try {
            // 从DOM工厂获得DOM解析器
            DocumentBuilder db = dbf.newDocumentBuilder();
            // 解析XML文档，得到一个Document对象，即DOM树
            Document doc = db.parse("CH02/src/添加收藏信息.xml");

            // 创建 Brand 节点
            Element brand = doc.createElement("Brand");
            brand.setAttribute("name", "三星");

            // 创建 Type 节点
            Element type = doc.createElement("Type");
            type.setAttribute("name", "NodeX");

            // 添加父子关系
            Text textNTT = doc.createTextNode("\n\t\t");
            brand.appendChild(textNTT);
            brand.appendChild(type);
            Text textNT = doc.createTextNode("\n\t");
            brand.appendChild(textNT);

            // 获得根节点，将创建的节点添加进来
            Element phoneInfo = (Element) doc.getElementsByTagName("PhoneInfo").item(0);

            // 向根元素添加子元素

            // 文本节点（制表符）
            Text textT = doc.createTextNode("\t");
            phoneInfo.appendChild(textT);
            phoneInfo.appendChild(brand);
            // 文本节点（换行）
            Text textN = doc.createTextNode("\n");
            phoneInfo.appendChild(textN);

            // 保存XML文件
            TransformerFactory transformerFactory = TransformerFactory.newInstance();
            Transformer transformer = transformerFactory.newTransformer();
            // 设置编码方式
            transformer.setOutputProperty(OutputKeys.ENCODING, "UTF-8");
            // 实例化DOMSource对象
            DOMSource source = new DOMSource(doc);
            // 实例化输出目标对象
            StreamResult result = new StreamResult(new FileOutputStream("CH02/src/添加收藏信息.xml"));
            // 将内存中的DOM保存到磁盘
            transformer.transform(source, result);

            System.out.println("添加成功！");
        } catch (ParserConfigurationException e) {
            e.printStackTrace();
        } catch (SAXException e) {
            e.printStackTrace();
        } catch (IOException e) {
            e.printStackTrace();
        } catch (TransformerConfigurationException e) {
            e.printStackTrace();
        } catch (TransformerException e) {
            e.printStackTrace();
        }
    }
}
